import numpy as np
import cv2

# 创建一个大小为 224 * 224 的二维矩阵，值为0~225之间
two_matrix = np.random.randint(0,256,(224,224),dtype=np.uint8)
print(two_matrix)


# 创建一个大小为 3 * 224 * 224 的三维矩阵，值为0~225之间
three_matrix = np.random.randint(0,256,(3,224,224),dtype=np.uint8)

# 在openCV中，图像的通道顺序是 高 * 宽 * 通道数
three_d_matrix = three_matrix.transpose(1,2,0)
print(three_d_matrix)

# 将矩阵转变成图像
cv2.imshow("two_matrix",two_matrix)
cv2.imshow("three_matrix",three_d_matrix)
cv2.waitKey(0)

#保存本地
cv2.imwrite("two_matrix.jpg",two_matrix)
cv2.imwrite("three_matrix.jpg",three_d_matrix)